package com.wxl.ruiji.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.wxl.ruiji.common.R;
import com.wxl.ruiji.dto.SetmealDto;
import com.wxl.ruiji.entity.Category;
import com.wxl.ruiji.entity.Setmeal;
import com.wxl.ruiji.service.CategoryService;
import com.wxl.ruiji.service.SetmealDishService;
import com.wxl.ruiji.service.SetmealService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.stream.Collectors;

@Slf4j
@RestController
@Api(tags = "套餐管理")
@RequestMapping("/setmeal")
public class SetmealController {

    @Autowired
    private SetmealService setmealService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private SetmealDishService setmealDishService;

    /**
     * 新增套餐
     *
     * @param setmealDto
     * @return
     */
    @ApiOperation(value = "新增套餐")
    @PostMapping
    public R<String> save(@RequestBody SetmealDto setmealDto) {

        log.info("新增套餐信息 => setmealDto：{}", setmealDto);

        setmealService.saveWithDish(setmealDto);

        return R.success("新增套餐成功！");

    }

    /**
     * 套餐分页查询
     *
     * @param page
     * @param pageSize
     * @param name
     * @return
     */
    @ApiOperation(value = "套餐分页查询")
    @GetMapping("/page")
    public R<Page> page(int page, int pageSize, String name) {

        log.info("套餐分页查询 => page: {}, pageSize: {}, name: {}", page, pageSize, name);

        // 分页构造器对象
        Page<Setmeal> pageInfo = new Page<>(page, pageSize);
        Page<SetmealDto> dtoPage = new Page<>();

        LambdaQueryWrapper<Setmeal> queryWrapper = new LambdaQueryWrapper<>();
        // 添加查询条件，根据 name 进行模糊查询
        queryWrapper.like(name != null, Setmeal::getName, name);
        // 添加排序条件，根据更新时间降序排列
        queryWrapper.orderByDesc(Setmeal::getUpdateTime);

        setmealService.page(pageInfo, queryWrapper);

        // 对象拷贝
        BeanUtils.copyProperties(pageInfo, dtoPage, "records");
        List<Setmeal> records = pageInfo.getRecords();

        List<SetmealDto> list = records.stream().map((item) -> {
            SetmealDto setmealDto = new SetmealDto();
            // 对象拷贝
            BeanUtils.copyProperties(item, setmealDto);
            // 分类 id
            Long categoryId = item.getCategoryId();
            // 根据分类 id 查询分类对象
            Category category = categoryService.getById(categoryId);
            if (category != null) {
                // 分类名称
                String categoryName = category.getName();
                setmealDto.setCategoryName(categoryName);
            }

            return setmealDto;

        }).collect(Collectors.toList());

        dtoPage.setRecords(list);

        return R.success(dtoPage);

    }

    /**
     * 删除套餐
     *
     * @param ids
     * @return
     */
    @ApiOperation(value = "删除套餐")
    @DeleteMapping
    public R<String> delete(@RequestParam List<Long> ids) {

        log.info("删除套餐 => ids: {}", ids);

        setmealService.removeWithDish(ids);

        return R.success("套餐数据删除成功！");

    }

    /**
     * 根据条件查询套餐数据
     *
     * @param setmeal
     * @return
     */
    @ApiOperation(value = "查询套餐数据")
    @GetMapping("/list")
    public R<List<Setmeal>> list(Setmeal setmeal) {

        log.info("根据条件查询套餐数据 => setmeal: {}", setmeal);

        LambdaQueryWrapper<Setmeal> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(setmeal.getCategoryId() != null, Setmeal::getCategoryId, setmeal.getCategoryId());
        queryWrapper.eq(setmeal.getStatus() != null, Setmeal::getStatus, setmeal.getStatus());
        queryWrapper.orderByDesc(Setmeal::getUpdateTime);

        List<Setmeal> list = setmealService.list(queryWrapper);

        return R.success(list);

    }

}
